FROM nvidia/cuda:9.0-runtime
LABEL maintainer = "<NVIDIA CORPORATION <cudatools@nvidia.com>"

############
# CUDNN
############

RUN echo "deb https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list

ENV CUDNN_VERSION 7.0.5.15
LABEL com.nvidia.cudnn.version="${CUDNN_VERSION}"

RUN apt-get update && apt-get install -y --no-install-recommends \
            libcudnn7=$CUDNN_VERSION-1+cuda9.0 && \
    apt-mark hold libcudnn7 && \
    rm -rf /var/lib/apt/lists/*

############
# KERAS & TF
############

ENV PYTHONUNBUFFERED 1

RUN apt-get update && apt-get install -y \
  python \
  python-pip \
  git

RUN pip install --upgrade pip

# Add source code
RUN mkdir /code
WORKDIR /code
ADD . /code/

# Add external libs
RUN mkdir /git_libs
RUN mkdir /tmp_imgs


#Keras
RUN git clone --branch 2.0.8.1 https://github.com/MarcBS/keras.git /git_libs/keras-smart-tray_v1.0

###Other
#RUN git clone https://github.com/lvapeab/coco-caption /git_libs/pycocoeval

# Add requirements
RUN pip install -r requirements.txt


